package libra;

import java.util.List;
import java.util.concurrent.ThreadLocalRandom;

import com.netflix.client.config.IClientConfig;
import com.netflix.loadbalancer.AbstractLoadBalancerRule;
import com.netflix.loadbalancer.ILoadBalancer;
import com.netflix.loadbalancer.Server;

/**
 * 自定义负载均衡，方式一
 * 		基于Ribbon替换负载均衡策略的方式，实现自定义负载均衡算法实现
 * 		参照RandomRule编写，规则：固定访问一台服务器5次后，切换至下一台服务器调用。
 * 
 * Ribbon的负载均衡算法：
 * 		rest接口第n次请求 % 服务器集群数m = 实际调用服务器下标index【从0开始】，每次服务重启后，接口计数从1开始。
 * 
核心组件IRule
	1、IRule：根据特定算法从服务列表中选取一个要访问的服务
	2、内置负载均衡方式
		2.1、RoundRobinRule
				轮序
		2.2、RandomRule
				随机
		2.3、AvailabilityFilteringRule
				会先过滤掉由于多次访问故障而处于断路器跳闸状态的服务，还有并发连接数超过阈值的服务
				然后对剩余的服务列表按照轮询策略进行访问
		2.4、WeightedResponseTimeRule
				根据平均响应时间计算所有服务的权重
				响应时间越快服务权重越大，被选中的概率越高
				刚启动时如果统计信息不足，会使用RandomRule策略
				等到统计信息充足后，会切换到WeightedResponseTimeRule
		2.5、RetryRule
				先按照RandomRule策略获取服务，如果获取服务失败，则在指定时间内进行重试，获取可用服务
		2.6、BestAvailableRule
				会先过滤掉由于多次访问故障处于断路器跳闸状态的服务，然后选择一个并发量最小的服务
		2.7、ZoneAvoidanceRule
				默认规则，复合判断server所在区域的性能和server的可用性来进行服务获取
 */

public class MyselfRandomIRule extends AbstractLoadBalancerRule {
	int total = 0;//当total==5以后，server的索引指针才会往下走
	int index = 0;//当前对外提供服务的服务器地址
    /**
     * Randomly choose from all living servers
     */
    public Server choose(ILoadBalancer lb, Object key) {
        if (lb == null) {
            return null;
        }
        Server server = null;

        while (server == null) {
            if (Thread.interrupted()) {
                return null;
            }
            List<Server> upList = lb.getReachableServers();
            List<Server> allList = lb.getAllServers();

            int serverCount = allList.size();
            
            if( total < 5 ) {
            	server = upList.get(index);
            	total++;
            	System.out.println( "MyselfRandomIRule[自定义负载策略]：当前访问次数：" + total + ", 当前服务下标："+index+"，当前服务："+server );
            } else {
            	total = 0;
            	index++;
            	if( index >= serverCount ) {
            		index = 0;
            	}
            }
            
//            if (serverCount == 0) {
//                /*
//                 * No servers. End regardless of pass, because subsequent passes
//                 * only get more restrictive.
//                 */
//                return null;
//            }
//
//            int index = chooseRandomInt(serverCount);
//            server = upList.get(index);
            
            
            if (server == null) {
                /*
                 * The only time this should happen is if the server list were
                 * somehow trimmed. This is a transient condition. Retry after
                 * yielding.
                 */
                Thread.yield();
                continue;
            }

            if (server.isAlive()) {
                return (server);
            }

            // Shouldn't actually happen.. but must be transient or a bug.
            server = null;
            Thread.yield();
        }

        return server;

    }

    protected int chooseRandomInt(int serverCount) {
        return ThreadLocalRandom.current().nextInt(serverCount);
    }

	@Override
	public Server choose(Object key) {
		return choose(getLoadBalancer(), key);
	}

	@Override
	public void initWithNiwsConfig(IClientConfig clientConfig) {
		// TODO Auto-generated method stub
	}
}
